package initialize

import (
	"demogo/global"
	"demogo/model"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"os"
)

func GormMysql() *gorm.DB {
	global.ASW_CONFIG.Mysql.Username = "root"
	global.ASW_CONFIG.Mysql.Password = "xiodi.cn123"
	global.ASW_CONFIG.Mysql.Dbname = "demogo"
	global.ASW_CONFIG.Mysql.Path = "127.0.0.1"
	global.ASW_CONFIG.Mysql.MaxOpenConns = 100     //设置数据库连接池最大连接数
	global.ASW_CONFIG.Mysql.MaxIdleConns = 20 //连接池最大允许的空闲连接数，如果没有sql任务需要执行的连接数大于20，超过的连接会被连接池关闭。
	global.ASW_CONFIG.Mysql.Config = "charset=utf8&parseTime=True&loc=Local"

	user := os.Getenv("MYSQL_USER")
	pwd := os.Getenv("MYSQL_PWD")
	host := os.Getenv("MYSQL_HOST")
	db := os.Getenv("MYSQL_DB")
    if len(user) != 0 {
		global.ASW_CONFIG.Mysql.Username = user
	}
	if len(pwd) != 0 {
		global.ASW_CONFIG.Mysql.Password = pwd
	}
	if len(host) != 0 {
		global.ASW_CONFIG.Mysql.Path = host
	}
	if len(db) != 0 {
		global.ASW_CONFIG.Mysql.Dbname = db
	}

	m := global.ASW_CONFIG.Mysql
	if m.Dbname == "" {
		return nil
	}
	dsn := m.Username + ":" + m.Password + "@tcp(" + m.Path + ")/" + m.Dbname + "?" + m.Config
	mysqlConfig := mysql.Config {
		DSN:                       dsn,   // DSN data source name
		DefaultStringSize:         191,   // string 类型字段的默认长度
		DisableDatetimePrecision:  true,  // 禁用 datetime 精度，MySQL 5.6 之前的数据库不支持
		DontSupportRenameIndex:    true,  // 重命名索引时采用删除并新建的方式，MySQL 5.7 之前的数据库和 MariaDB 不支持重命名索引
		DontSupportRenameColumn:   true,  // 用 `change` 重命名列，MySQL 8 之前的数据库和 MariaDB 不支持重命名列
		SkipInitializeWithVersion: false, // 根据版本自动配置
	}
	if db, err := gorm.Open(mysql.New(mysqlConfig)); err != nil {
		panic(err)
	} else {
		sqlDB, _ := db.DB()
		sqlDB.SetMaxIdleConns(m.MaxIdleConns)
		sqlDB.SetMaxOpenConns(m.MaxOpenConns)
		return db
	}
}

func MysqlTables(db *gorm.DB) {
	// 自动初始化表
	err := db.AutoMigrate(
		model.User{},
	)
	if err != nil {
		os.Exit(0)
	}
}
